//
//  StatusPictureView.swift
//  微博009
//
//  Created by Romeo on 15/9/7.
//  Copyright © 2015年 itheima. All rights reserved.
//

import UIKit
import SDWebImage

/// 可重用标识符
private let HMStatusPictureViewCellID = "HMStatusPictureViewCellID"

class StatusPictureView: UICollectionView {

    /// 微博数据视图模型 － 如果是新的 cell 才会设置模型
    var statusViewModel: StatusViewModel? {
        didSet {
            sizeToFit()
            
            // 刷新数据
            reloadData()
        }
    }
    
    override func sizeThatFits(size: CGSize) -> CGSize {
        return calcViewSize()
    }
    
    /// 根据模型中的图片数量来计算视图大小
    private func calcViewSize() -> CGSize {
        // 1. 准备工作
        let layout = collectionViewLayout as! UICollectionViewFlowLayout
        // 设置默认大小
        layout.itemSize = CGSize(width: HMStatusPictureItemWidth, height: HMStatusPictureItemWidth)
        
        // 2. 根据图片数量来计算大小
        let count = statusViewModel?.thumbnailURLs?.count ?? 0
        
        // 1> 没有图
        if count == 0 {
            return CGSizeZero
        }
        
        // 2> 1张图
        if count == 1 {
            var size = CGSize(width: 150, height: 150)
            
            // 判断图片是否已经被正确的缓存 Key 是 URL 的完整字符串
            let key = statusViewModel!.thumbnailURLs![0].absoluteString
            
            // 如果有缓存图片，记录当前图片的大小
            if let image = SDWebImageManager.sharedManager().imageCache.imageFromDiskCacheForKey(key) {
                size = image.size
            }
            
            // 单独处理过宽或者过窄的图片
            size.width = size.width < 40 ? 40 : size.width
            size.width = size.width > 300 ? 300 : size.width
            
            layout.itemSize = size
            return size
        }
        
        // 3> 4张图 2 * 2 
        if count == 4 {
            let w = 2 * HMStatusPictureItemWidth + HMStatusPictureItemMargin
            
            return CGSize(width: w, height: w)
        }
        
        // 4> 其他
        /**
            2, 3,
            5, 6,
            7, 8, 9
        */
        // 计算显示图片的行数
        let row = CGFloat((count - 1) / Int(HMStatusPictureMaxCount) + 1)
        let h = row * HMStatusPictureItemWidth + (row - 1) * HMStatusCellMargin
        let w = HMStatusPictureMaxWidth
        
        return CGSize(width: w, height: h)
    }
    
    // 构造函数的调用是底层自动转发的 init() -> initWithFrame -> initWithFrame:layout:
    // 默认的 layout 没有被初始化
    override init(frame: CGRect, collectionViewLayout layout: UICollectionViewLayout) {
        super.init(frame: frame, collectionViewLayout: UICollectionViewFlowLayout())
        
        backgroundColor = UIColor.lightGrayColor()
        
        // 设置布局的间距
        let layout = collectionViewLayout as! UICollectionViewFlowLayout
        layout.minimumInteritemSpacing = HMStatusPictureItemMargin
        layout.minimumLineSpacing = HMStatusPictureItemMargin
        
        // 指定数据源 － 让自己当自己的数据源
        // 1. 在自定义 view 中，代码逻辑相对简单，可以考虑自己充当自己的数据源
        // 2. dataSource & delegate 本身都是弱引用，自己充当自己的代理不会产生循环引用
        // 3. 除了配图视图，自定义 pickerView(省市联动的)
        dataSource = self
        // 注册可重用 cell
        registerClass(StatusPictureViewCell.self, forCellWithReuseIdentifier: HMStatusPictureViewCellID)
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}

extension StatusPictureView: UICollectionViewDataSource {
    
    func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return statusViewModel?.thumbnailURLs?.count ?? 0
    }
    
    func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
        
        let cell = collectionView.dequeueReusableCellWithReuseIdentifier(HMStatusPictureViewCellID, forIndexPath: indexPath) as! StatusPictureViewCell
        
        cell.imageURL = statusViewModel!.thumbnailURLs![indexPath.item]

        return cell
    }
}

/// 配图视图的 Cell
private class StatusPictureViewCell: UICollectionViewCell {
    
    /// 配图视图的 URL
    var imageURL: NSURL? {
        didSet {
            iconView.sd_setImageWithURL(imageURL)
        }
    }
    
    // MARK: - 构造函数
    override init(frame: CGRect) {
        super.init(frame: frame)
        
        addSubview(iconView)
        iconView.ff_Fill(self)
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    // MARK: - 懒加载
    private let iconView: UIImageView = {
       
        let iv = UIImageView()
        
        // 设置填充模式
        iv.contentMode = UIViewContentMode.ScaleAspectFill
        iv.clipsToBounds = true
        
        return iv
    }()
}
